# -*- coding:UTF-8 -*-

'''
    多线程
'''
import log_utils as log

# -----------------     _thread 使用     -----------------

log.loge("_thread 使用")
import time


# 为纯种定义一个函数
def print_time(threadName, delay):
    count = 0
    while count < 5:
        time.sleep(delay)
        count += 1
        print("%s: %s" % (threadName, time.ctime(time.time())))


# 创建两个线程
# _thread.start_new(print_time, ("Thread-1", 2,))
# _thread.start_new(print_time, ("Thread-2", 4,))
#
# while 1:
#     pass
log.loge("_thread 使用")
# -----------------     _thread 使用     -----------------


# -----------------     threading 使用     -----------------
"""
使用 threading 模块创建线程
我们可以通过直接从 threading.Thread 继承创建一个新的子类，并实例化后调用 start() 方法启动新线程，即它调用了线程的 run() 方法：
"""
log.loge("threading 使用")
import threading

exitFlag = 0


class myThread(threading.Thread):
    def __init__(self, threadID, name, counter):
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.name = name
        self.counter = counter

    def run(self):
        print("开始线程: ", self.name)
        print_time(self.name, self.counter, 5)
        print("退出线程: ", self.name)


def print_time(threadName, delay, counter):
    while counter:
        if exitFlag:
            threadName.exit()
        time.sleep(delay)
        print("%s: %s" % (threadName, time.ctime(time.time())))
        counter -= 1


# 创建新线程
thread1 = myThread(1, "Thread-1", 1)
thread2 = myThread(2, "Thread-2", 2)

# 开启新线程
thread1.start()
thread2.start()

thread1.join()
thread2.join()

print("退出主线程")
log.loge("threading 使用")
# -----------------     threading 使用     -----------------


# -----------------     线程同步 < threadLock安装异常，无法 测试 >    -----------------
# import threadLock
# #
# #
# class mySyncThread(threading.Thread):
#     def __init__(self, threadID, name, counter):
#         threading.Thread.__init__(self)
#         self.threadID = threadID
#         self.name = name
#         self.counter = counter
#
#     def run(self):
#         print("开启线程: ", self.name)
#         print(self.name, "获取锁，用于线程同步")
#
#         # threadLock 需要安装
#         threadLock.acquire()
#
#         print_time(self.name, self.counter, 5)
#
#         # 释放锁,开启下一个线程
#         threadLock.release()
#
#
# threadLock = threading.Lock()
# threads = []
# #
# # 创建新线程
# thread1 = mySyncThread(1, "Thread-1", 1)
# thread2 = mySyncThread(2, "Thread-2", 2)
#
# # 开启新线程
# thread1.start()
# thread2.start()
#
# # 添加线程到线程列表
# threads.append(thread1)
# threads.append(thread2)
#
# # 等待所有线程完成
# for t in threads:
#     t.join()
# print("退出主线程")
# -----------------     线程同步  < threadLock安装异常，无法 测试 >     -----------------


# -----------------     线程优先级队列（ Queue）     -----------------
import queue

log.loge("线程优先级队列（ Queue）")
exitFlag = 0


class myThread(threading.Thread):
    def __init__(self, threadID, name, q):
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.name = name
        self.q = q

    def run(self):
        print("开启线程：" + self.name)
        process_data(self.name, self.q)
        print("退出线程：" + self.name)


def process_data(threadName, q):
    while not exitFlag:
        queueLock.acquire()
        if not workQueue.empty():
            data = q.get()
            queueLock.release()
            print("%s processing %s" % (threadName, data))
        else:
            queueLock.release()
        time.sleep(1)


threadList = ["Thread-1", "Thread-2", "Thread-3"]
nameList = ["One", "Two", "Three", "Four", "Five"]
queueLock = threading.Lock()
workQueue = queue.Queue(10)
threads = []
threadID = 1

# 创建新线程
for tName in threadList:
    thread = myThread(threadID, tName, workQueue)
    thread.start()
    threads.append(thread)
    threadID += 1

# 填充队列
queueLock.acquire()
for word in nameList:
    workQueue.put(word)
queueLock.release()

# 等待队列清空
while not workQueue.empty():
    pass

# 通知线程是时候退出
exitFlag = 1

# 等待所有线程完成
for t in threads:
    t.join()
print("退出主线程")
log.loge("线程优先级队列（ Queue）")
# -----------------     线程优先级队列（ Queue）     -----------------
